import datastructure.binarytree.TreeNode;

/*
 * @lc app=leetcode.cn id=236 lang=java
 *
 * [236] 二叉树的最近公共祖先
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 * int val;
 * TreeNode left;
 * TreeNode right;
 * TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null) {
            return null;
        }
        // 如果当前节点等于 p 或者 q，意味着当前节点就是公共祖先。
        if (root.val == p.val || root.val == q.val) {
            return root;
        }
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        // 如果当前节点的左右子树分别找到了等于 p 或者 q 的节点，则当前节点是公共祖先。
        if (left != null && right != null) {
            return root;
        }
        // 否则，有一个节点本身是另一个节点的祖先。
        return left != null ? left : right;

    }

}
// @lc code=end
